#include <iostream>
#include <string>
#include <sstream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
using namespace std;

#include "../parteehparticles/particle.hpp"
#include "../parteehparticles/particleEmitter.hpp"
#include "../parteehparticles/particleSystem.hpp"
#include "../parteehparticles/particleAttractor.hpp"
#include "../parteehparticles/our_time.h"


#include <windows.h>
#include <gl/gl.h>

LRESULT CALLBACK WindowProc(HWND, UINT, WPARAM, LPARAM);
void EnableOpenGL(HWND hwnd, HDC*, HGLRC*);
void DisableOpenGL(HWND, HDC, HGLRC);


int WINAPI WinMain(HINSTANCE hInstance,
                   HINSTANCE hPrevInstance,
                   LPSTR lpCmdLine,
                   int nCmdShow) {
    WNDCLASSEX wcex;
    HWND hwnd;
    HDC hDC;
    HGLRC hRC;
    MSG msg;
    BOOL bQuit = FALSE;

    /* register window class */
    wcex.cbSize = sizeof(WNDCLASSEX);
    wcex.style = CS_OWNDC;
    wcex.lpfnWndProc = WindowProc;
    wcex.cbClsExtra = 0;
    wcex.cbWndExtra = 0;
    wcex.hInstance = hInstance;
    wcex.hIcon = LoadIcon(NULL, IDI_APPLICATION);
    wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
    wcex.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);
    wcex.lpszMenuName = NULL;
    wcex.lpszClassName = "GLSample";
    wcex.hIconSm = LoadIcon(NULL, IDI_APPLICATION);;


    if (!RegisterClassEx(&wcex))
        return 0;

    /* create main window */
    hwnd = CreateWindowEx(0,
                          "GLSample",
                          "OpenGL Sample",
                          WS_OVERLAPPEDWINDOW,
                          CW_USEDEFAULT,
                          CW_USEDEFAULT,
                          640,
                          640,
                          NULL,
                          NULL,
                          hInstance,
                          NULL);

    ShowWindow(hwnd, nCmdShow);

    /* enable OpenGL for the window */
    EnableOpenGL(hwnd, &hDC, &hRC);
    glOrtho(0, 200, -100, 30, -100, 100);

    /* Init particles */
    ParticleSystem world;
    world.set_gravity(1);
    world.max_particles = 20000;

    ParticleEmitter *emit = new ParticleEmitter(&world,
            100, 0, //x, y
            180, 180, //angle
            4, .5,  //power
            3, 2, //scale
            4, 8, //rotation_speed
            5, 0, //mass
            1, .1, //dt_scale
            1000);

    world.add_emitter(emit);

    glClearColor(0.0f, 0.0f, 0.0f, 0.0f);
    glClear(GL_COLOR_BUFFER_BIT);
    SwapBuffers(hDC);
    glClear(GL_COLOR_BUFFER_BIT);

    /* program main loop */
    while (!bQuit) {
        /* check for messages */
        if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
            /* handle or dispatch messages */
            if (msg.message == WM_QUIT) {
                bQuit = TRUE;
            } else {
                TranslateMessage(&msg);
                DispatchMessage(&msg);
            }
        } else {
            //cout << get_seconds<float>() << endl;
            /* OpenGL animation code goes here */
            //RenderFrame();

            //glClearColor(0.0f, 0.0f, 0.0f, 0.1f);
            //glClear(GL_COLOR_BUFFER_BIT);
            for ( vector<Particle*>::iterator it = world.particles.begin(); it != world.particles.end(); it++ ) {
#ifdef HARDDEBUG
                world.emitters[0]->output();
                (*it)->output();
                cin.ignore();
#endif

                glPushMatrix();
                glTranslatef((*it)->x,(*it)->y,0);
                glRotatef((*it)->rotation, 0.0f, 0.0f, 1.0f);
                glScalef((*it)->scale,(*it)->scale,1.0f);
                float kleur = sin((*it)->x_speed*20)*.5+.5;
                float kleur2 = sin((*it)->y_speed*20)*.5+.5;
                float kleur3 = cos((*it)->ID*0.000)*.5+.5;

                glBegin(GL_TRIANGLES);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(0.0f,   1.0f);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(0.87f,  -0.5f);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(-0.87f, -0.5f);
                glEnd();

                glRotatef(180, 0.0f, 0.0f, 1.0f);

                glBegin(GL_TRIANGLES);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(0.0f,   1.0f);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(0.87f,  -0.5f);
                glColor3f(kleur3, kleur, kleur2);
                glVertex2f(-0.87f, -0.5f);
                glEnd();

                glPopMatrix();

            }

            SwapBuffers(hDC);

            world.update();

            Sleep(50);
        }
    }

    /* shutdown OpenGL */
    DisableOpenGL(hwnd, hDC, hRC);

    /* destroy the window explicitly */
    DestroyWindow(hwnd);

    return msg.wParam;
}

LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch (uMsg) {
    case WM_CLOSE:
        PostQuitMessage(0);
        break;

    case WM_DESTROY:
        return 0;

    case WM_KEYDOWN: {
        switch (wParam) {
        case VK_ESCAPE:
            PostQuitMessage(0);
            break;
        }
    }
    break;

    default:
        return DefWindowProc(hwnd, uMsg, wParam, lParam);
    }

    return 0;
}

void EnableOpenGL(HWND hwnd, HDC* hDC, HGLRC* hRC) {
    PIXELFORMATDESCRIPTOR pfd;

    int iFormat;

    /* get the device context (DC) */
    *hDC = GetDC(hwnd);

    /* set the pixel format for the DC */
    ZeroMemory(&pfd, sizeof(pfd));

    pfd.nSize = sizeof(pfd);
    pfd.nVersion = 1;
    pfd.dwFlags = PFD_DRAW_TO_WINDOW |
                  PFD_SUPPORT_OPENGL | PFD_DOUBLEBUFFER;
    pfd.iPixelType = PFD_TYPE_RGBA;
    pfd.cColorBits = 24;
    pfd.cDepthBits = 16;
    pfd.iLayerType = PFD_MAIN_PLANE;

    iFormat = ChoosePixelFormat(*hDC, &pfd);

    SetPixelFormat(*hDC, iFormat, &pfd);

    /* create and enable the render context (RC) */
    *hRC = wglCreateContext(*hDC);

    wglMakeCurrent(*hDC, *hRC);
}

void DisableOpenGL (HWND hwnd, HDC hDC, HGLRC hRC) {
    wglMakeCurrent(NULL, NULL);
    wglDeleteContext(hRC);
    ReleaseDC(hwnd, hDC);
}


